package com.yc;

import com.alibaba.csp.sentinel.Entry;
import com.alibaba.csp.sentinel.SphU;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;

import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName a
 * @since: 2023/8/25 14:12
 * @auth: kirito
 * @description:
 **/
public class a {
    public static void main(String[] args) {
        initFlowRules();
        while (true) {
            Entry entry = null;
            try {
                entry = SphU.entry("HelloWorld");
                /*您的业务逻辑 - 开始*/
                System.out.println("hello world");
                /*您的业务逻辑 - 结束*/
            } catch (BlockException e1) {
                /*流控逻辑处理 - 开始*/
                System.out.println("block!");
                /*流控逻辑处理 - 结束*/
            } finally {
                if (entry != null) {
                    entry.exit();
                }
            }
        }
    }

    /**
     * initFlowRules()方法用于初始化流控规则。在示例中，创建了一个针对资源名为"HelloWorld"的流控规则，限制每秒请求数不超过2000。
     *
     * 在main方法中，通过SphU.entry("HelloWorld")获取资源的入口，并执行相应的业务逻辑。在示例中，输出了"hello world"。
     *
     * 如果资源的访问超过了流控规则所定义的限制，将会抛出BlockException，在catch块中可以处理流控逻辑。在示例中，输出了"block!"。
     *
     * 无论是否发生了流控，都需要在finally块中调用entry.exit()来释放资源的入口。
     *
     * 请注意，此示例中的流控规则是针对资源名为"HelloWorld"的限制，您可以根据实际需求修改资源名和流控规则的设置。
     */
    private static void initFlowRules(){
        List<FlowRule> rules = new ArrayList<>();
        FlowRule rule = new FlowRule();
        rule.setResource("HelloWorld");
        rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
        // Set limit QPS to 20.
        rule.setCount(2000);
        rules.add(rule);
        FlowRuleManager.loadRules(rules);
    }
}
